# 查核酸
def check():
    print("欢迎来到黑马程序员！")
    print("请出示您健康码以及72小时核酸证明！")


check()
# 黑马ATM
mony = 50000
name = input("请输入您的姓名")


def main():
    print("----------主菜单----------")
    print(f"{name}您好，欢迎来到黑马ATM请选择您的操作")
    print("查询余额 [输入1]")
    print("存款    [输入2]")
    print("取款    [输入3]")
    print("退出    [输入4]")
    flag = int(input("请输入您的选择"))
    if flag == 1:
        fun1()
    elif flag == 2:
        fun2()
    elif flag == 3:
        fun3()
    else:
        print("欢迎下次光临，再见")
        return None


def fun1():
    print("----------查询余额----------")
    print(f"{name},您好,您的余额剩余{mony}元")
    main()


def fun2():
    print("----------存款----------")
    my = int(input("请输入人您要存款的金额"))
    global mony
    mony+=my
    print(f"{name},您好,存款{my}元成功,您的余额剩余{mony}元")
    main()

def fun3():
    print("----------取款----------")
    my = int(input("请输入人您要取款的金额"))
    global mony
    if mony < my:
        print("取款失败，您的余额不足")
    else:
        mony -= my
        print(f"{name},您好,取款{my}元成功,您的余额剩余{mony}元")
    main()

main()